package leetcode;

import java.util.LinkedList;

/**
 * 394. 字符串解码
 * 给定一个经过编码的字符串，返回它解码后的字符串。
 * <p>
 * 编码规则为: k[encoded_string]，表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。
 * <p>
 * 你可以认为输入字符串总是有效的；输入字符串中没有额外的空格，且输入的方括号总是符合格式要求的。
 * <p>
 * 此外，你可以认为原始数据不包含数字，所有的数字只表示重复的次数 k ，例如不会出现像 3a 或 2[4] 的输入。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：s = "3[a]2[bc]"
 * 输出："aaabcbc"
 * 示例 2：
 * <p>
 * 输入：s = "3[a2[c]]"
 * 输出："accaccacc"
 * 示例 3：
 * <p>
 * 输入：s = "2[abc]3[cd]ef"
 * 输出："abcabccdcdcdef"
 * 示例 4：
 * <p>
 * 输入：s = "abc3[cd]xyz"
 * 输出："abccdcdcdxyz"
 */
public class DecodeString {


    /**
     * 栈
     *
     * @param s
     * @return
     */
    public String decodeString(String s) {
        StringBuilder ans = new StringBuilder();
        int num = 0;

        LinkedList<Integer> numStack = new LinkedList<>();
        LinkedList<String> ansStack = new LinkedList<>();

        for (Character c : s.toCharArray()) {
            if (c == '[') {

                // 把前一个结果和次数存到栈里
                numStack.addLast(num);// 下下次']'会用到
                ansStack.addLast(ans.toString()); // 下次']'会用到

                // num 和 ans 归零
                num = 0;
                ans = new StringBuilder();
            } else if (c == ']') {

                // 往 ans 上拼接过
                StringBuilder tmp = new StringBuilder();
                int curNum = numStack.removeLast();
                // curNum 个 ans
                for (int i = 0; i < curNum; i++) {

                    tmp.append(ans);
                }
                // ans 不如栈，如果后面有'['再入栈
                ans = new StringBuilder(ansStack.removeLast() + tmp);
            } else if (c >= '0' && c <= '9') {

                // 记录数字 可能是多位数字
                num = num * 10 + Integer.parseInt(c + "");
            } else {

                // 记录字母，可能上一个字符是],已经拼过数字了
                ans.append(c);
            }
        }
        return ans.toString();
    }

}
